/* -*- c++ -*- */
/*
 * Copyright 2009-2014 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * GNU Radio is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#ifndef INCLUDED_THREAD_H
#define INCLUDED_THREAD_H

#include <gnuradio/api.h>
#include <boost/thread/thread.hpp>
#include <boost/thread/mutex.hpp>
#include <boost/thread/locks.hpp>
#include <boost/thread/condition_variable.hpp>
#include <boost/thread/barrier.hpp>
#include <boost/shared_ptr.hpp>
#include <vector>

#if defined(_WIN32) || defined(__WIN32__) || defined(WIN32)

#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif

#include <windows.h>

#endif

namespace gr {
  namespace thread {

    typedef boost::thread                    thread;
    typedef boost::mutex                     mutex;
    typedef boost::unique_lock<boost::mutex> scoped_lock;
    typedef boost::condition_variable        condition_variable;
    typedef boost::barrier                   barrier;
    typedef boost::shared_ptr<barrier>       barrier_sptr;

    /*! \brief a system-dependent typedef for the underlying thread type.
     */
#if defined(_WIN32) || defined(__WIN32__) || defined(WIN32)
    typedef HANDLE gr_thread_t;
#elif defined(macintosh) || defined(__APPLE__) || defined(__APPLE_CC__)
    typedef pthread_t gr_thread_t;
#else
    typedef pthread_t gr_thread_t;
#endif

    /*! \brief Get the current thread's ID as a gr_thread_t
     *
     * We use this when setting the thread affinity or any other
     * low-level thread settings. Can be called withing a GNU Radio
     * block to get a reference to its current thread ID.
     */
    GR_RUNTIME_API gr_thread_t get_current_thread_id();

    /*! \brief Bind the current thread to a set of cores.
     *
     * Wrapper for system-dependent calls to set the affinity of the
     * current thread to the processor mask. The mask is simply a
     * 1-demensional vector containing the processor or core number
     * from 0 to N-1 for N cores.
     *
     * Note: this does not work on OSX; it is a nop call since OSX
     * does not support the concept of thread affinity (and what they
     * do support in this way since 10.5 is not what we want or can
     * use in this fashion).
     */
    GR_RUNTIME_API void thread_bind_to_processor(const std::vector<int> &mask);

    /*! \brief Convineince function to bind the current thread to a single core.
     *
     * Wrapper for system-dependent calls to set the affinity of the
     * current thread to a given core from 0 to N-1 for N cores.
     *
     * Note: this does not work on OSX; it is a nop call since OSX
     * does not support the concept of thread affinity (and what they
     * do support in this way since 10.5 is not what we want or can
     * use in this fashion).
     */
    GR_RUNTIME_API void thread_bind_to_processor(int n);

    /*! \brief Bind a thread to a set of cores.
     *
     * Wrapper for system-dependent calls to set the affinity of the
     * given thread ID to the processor mask. The mask is simply a
     * 1-demensional vector containing the processor or core number
     * from 0 to N-1 for N cores.
     *
     * Note: this does not work on OSX; it is a nop call since OSX
     * does not support the concept of thread affinity (and what they
     * do support in this way since 10.5 is not what we want or can
     * use in this fashion).
     */
    GR_RUNTIME_API void thread_bind_to_processor(gr_thread_t thread,
                                                 const std::vector<int> &mask);


    /*! \brief Convineince function to bind the a thread to a single core.
     *
     * Wrapper for system-dependent calls to set the affinity of the
     * given thread ID to a given core from 0 to N-1 for N cores.
     *
     * Note: this does not work on OSX; it is a nop call since OSX
     * does not support the concept of thread affinity (and what they
     * do support in this way since 10.5 is not what we want or can
     * use in this fashion).
     */
    GR_RUNTIME_API void thread_bind_to_processor(gr_thread_t thread,
                                                 unsigned int n);

    /*! \brief Remove any thread-processor affinity for the current thread.
     *
     * Note: this does not work on OSX; it is a nop call since OSX
     * does not support the concept of thread affinity (and what they
     * do support in this way since 10.5 is not what we want or can
     * use in this fashion).
     */
    GR_RUNTIME_API void thread_unbind();

    /*! \brief Remove any thread-processor affinity for a given thread ID.
     *
     * Note: this does not work on OSX; it is a nop call since OSX
     * does not support the concept of thread affinity (and what they
     * do support in this way since 10.5 is not what we want or can
     * use in this fashion).
     */
    GR_RUNTIME_API void thread_unbind(gr_thread_t thread);

    /*! \brief get current thread priority for a given thread ID
     */
    GR_RUNTIME_API int thread_priority(gr_thread_t thread);

    /*! \brief set current thread priority for a given thread ID
     */
    GR_RUNTIME_API int set_thread_priority(gr_thread_t thread, int priority);

    GR_RUNTIME_API void set_thread_name(gr_thread_t thread,
                                        std::string name);

  } /* namespace thread */
} /* namespace gr */

#endif /* INCLUDED_THREAD_H */
